package com.ks.utils;

import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.UnknownHostException;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.Set;

import javax.management.AttributeNotFoundException;
import javax.management.InstanceNotFoundException;
import javax.management.MBeanException;
import javax.management.MBeanServer;
import javax.management.MBeanServerFactory;
import javax.management.ObjectName;
import javax.management.ReflectionException;
/***
 * 获取 tomcat 本机的 IP 地址 以及 端口号
 * @author pks
 * @date 2019年9月10日
 */
public class SUtilService {
	
	 /**
     * 获取本地IP地址
     * @return
     * @throws UnknownHostException
     */
    public static String getHostAddress() {
    	try {
    		Enumeration<NetworkInterface> netInterfaces = null;
            netInterfaces = NetworkInterface.getNetworkInterfaces();
            while (netInterfaces.hasMoreElements()) {
                NetworkInterface ni = netInterfaces.nextElement();
                Enumeration<InetAddress> ips = ni.getInetAddresses();
                while (ips.hasMoreElements()) {
                    InetAddress ip = ips.nextElement();
                    if (ip.isSiteLocalAddress()) {
                        return ip.getHostAddress();
                    }
                }
            }
	        return InetAddress.getLocalHost().getHostAddress();
		} catch (Exception e) {
			e.printStackTrace();
		}
    	return "";
    }

    
    /**
     * 获取服务端口号
     * @return 端口号
     * @throws ReflectionException
     * @throws MBeanException
     * @throws InstanceNotFoundException
     * @throws AttributeNotFoundException
     */
    public static String getServerPort(boolean secure) {
        try {
        	MBeanServer mBeanServer = null;
            if (MBeanServerFactory.findMBeanServer(null).size() > 0) {
                mBeanServer = (MBeanServer)MBeanServerFactory.findMBeanServer(null).get(0);
            }
            
            if (mBeanServer == null) {
//                log.debug("调用findMBeanServer查询到的结果为null");
                return "";
            }
            
            Set<ObjectName> names = null;
            try {
                names = mBeanServer.queryNames(new ObjectName("Catalina:type=Connector,*"), null);
            } catch (Exception e) {
                return "";
            }
            Iterator<ObjectName> it = names.iterator();
            ObjectName oname = null;
            while (it.hasNext()) {
                oname = (ObjectName)it.next();
                String protocol = (String)mBeanServer.getAttribute(oname, "protocol");
                String scheme = (String)mBeanServer.getAttribute(oname, "scheme");
                Boolean secureValue = (Boolean)mBeanServer.getAttribute(oname, "secure");
                Boolean SSLEnabled = (Boolean)mBeanServer.getAttribute(oname, "SSLEnabled");
                if (SSLEnabled != null && SSLEnabled) {// tomcat6开始用SSLEnabled
                    secureValue = true;// SSLEnabled=true但secure未配置的情况
                    scheme = "https";
                }
                if (protocol != null && ("HTTP/1.1".equals(protocol) || protocol.contains("http"))) {
                    if (secure && "https".equals(scheme) && secureValue) {
                        return ((Integer)mBeanServer.getAttribute(oname, "port")).toString();
                    } else if (!secure && !"https".equals(scheme) && !secureValue) {
                        return ((Integer)mBeanServer.getAttribute(oname, "port")).toString();
                    }
                }
            }
		} catch (Exception e) {
			e.printStackTrace();
		}
        return "";
    }
}
